package com.lw.utils;

import org.springframework.web.multipart.MultipartFile;

import java.util.Arrays;
import java.util.List;

public class Utils {
    //1、设置上传文件类型(图片类型)（可以设置允许上传的图片格式,后面介绍分割前缀和后缀）
    private static String ALLOW_IMG_TYPE = "image/png;image/jpg;image/gif;image/jpeg";

    public boolean checkImage(MultipartFile file){
        //主要限制在前端上进行，后端做一些简单的限制
        List<String> typeList = Arrays.asList(ALLOW_IMG_TYPE.split(";")); //将数组转换成list集合
        if (file == null) {
            throw new RuntimeException("图片不能为空");
        }
        if (file.getSize() >= 20 * 1024 * 1024) {
            throw new RuntimeException("图片大小超出最大限制");
        }
        if(!typeList.contains(file.getContentType())){
            throw new RuntimeException("图片格式不符合");
        }
        return true;
    }
}
